from wenet.utils.init_model import init_model
from gxl_ai_utils.utils import utils_file
import argparse
config_path = './conf/train_u2++_moe_conformer.yaml'
# config_path = "D:\gengxuelong_code\wenet\examples\\aishell\s0\conf\\train_u2++_conformer.yaml"
configs = utils_file.load_dict_from_yaml(config_path)
args = argparse.Namespace()
model,configs = init_model(args, configs)
print(model)
utils_file.print_model_size(model.encoder)
utils_file.print_model_size(model.decoder)
utils_file.print_model_size(model.ctc)
